try:
    from .AIAssistant import AIAssistant  # type: ignore
except Exception:
    import os
    import sys
    sys.path.append(os.path.dirname(__file__))
    from AIAssistant import AIAssistant  # type: ignore


class ImageGenerationAssistant(AIAssistant):
    def __init__(self) -> None:
        super().__init__()
        from typing import Optional
        self.name: Optional[str] = None
        self.type: Optional[str] = None

    def painting(self) -> None:
        name_str = self.name or ""
        type_str = self.type or ""
        print(f"🎨{name_str}{type_str}正在为你作画，请稍等......")


